828A - Restaurant Tables - CodeForces Solution


implementation *1200

Please click on ads to support us..

Python Code:

n, a, b = map(int, input().split())
l = list(map(int, input().split()))
ans = 0
c = 0
for i in l:
    if i == 1:
        if a != 0:
            a -= i;ans += 1
        elif b != 0:
            ans += i;b -= 1;c += 1
        elif c != 0:
            ans += i;c -= 1
    else:
        if b != 0:
            ans += i;b -= 1
print(sum(l)-ans)

C++ Code:

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;

int main(){
    ll n,a,b;
    cin>>n>>a>>b;
    ll x=0;
    ll c=0;
    for(int i=0;i<n;++i){
        ll k;
        cin>>k;
        if(k==1){
            if(a>0){
                a--;
            }
            else if(b>0){
                b--;
                x++;
            }
            else if(x>0){
                x--;
            }
            else c++;
        }
        if(k==2){
            if(b>0){
                b--;
            } else c=c+2;
        }
    }
    cout<<c<<endl;
}


Comments

Submit
0 Comments
More Questions

281A - Word Capitalization
1646A - Square Counting
266A - Stones on the Table
61A - Ultra-Fast Mathematician
148A - Insomnia cure
1650A - Deletions of Two Adjacent Letters
1512A - Spy Detected
282A - Bit++
69A - Young Physicist
1651A - Playoff
734A - Anton and Danik
1300B - Assigning to Classes
1647A - Madoka and Math Dad
710A - King Moves
1131A - Sea Battle
118A - String Task
236A - Boy or Girl
271A - Beautiful Year
520B - Two Buttons
231A - Team
479C - Exams
1030A - In Search of an Easy Problem
158A - Next Round
71A - Way Too Long Words
160A - Twins
1A - Theatre Square
1614B - Divan and a New Project
791A - Bear and Big Brother
1452A - Robot Program
344A - Magnets